# -*- coding: utf-8 -*-

import numpy as np
import pandas as pd
import matplotlib as mpl
import matplotlib.pyplot as plt

# =======================
mpl.style.use('seaborn-whitegrid');


def sta001(k, nyear, xd):
    d2 = np.fv(k, nyear, -xd, -xd);
    d2 = round(d2)
    return d2


# =======================


d40 = 1.4 * 40
print("d40,40 x 1.4=", d40)
d = sta001(0.05, 40 - 1, 1.4);
print("01保守投资模式，", d, round(d / d40))

d2 = sta001(0.20, 40 - 1, 1.4);
print("02激进投资模式，", d2, round(d2 / d40))

dk = round(d2 / d)
print("dk,两者差别（xx倍）：", dk)

dx05 = [sta001(0.05, x, 1.4) for x in range(0, 40)]
dx10 = [sta001(0.10, x, 1.4) for x in range(0, 40)]
dx15 = [sta001(0.15, x, 1.4) for x in range(0, 40)]
dx20 = [sta001(0.20, x, 1.4) for x in range(0, 40)]
# print(dx05)print(dx20)

df = pd.DataFrame(columns=['dx05', 'dx10', 'dx15', 'dx20']);
df['dx05'] = dx05;
df['dx10'] = dx10;
df['dx15'] = dx15;
df['dx20'] = dx20;
print("")
print(df.tail())
df.plot();
